#include <stdio.h>
#include <string.h>

int main(){
	printf("Enter message to be encrypted: ");
	char msg[100];
	gets(msg);
	int shift;
	printf("Enter shift amount(1-25):");
	scanf("%d", &shift);

	for (int i = 0; msg[i] != '\0'; ++i){
		if (msg[i] <= 'z' && msg[i] >= 'a'){
			if (msg[i] + shift > 'z'){
				msg[i] = (msg[i] + shift) - 'z' + 'a' - 1;
			} else 
				msg[i] += shift;
		} else if (msg[i] <= 'Z' && msg[i] >= 'A'){
			if (msg[i] + shift > 'Z') 
				msg[i] = msg[i] + shift - 'Z' + 'A' - 1; 
			else
				msg[i] += shift;
		}
	}

	printf("Encrypted message: ");
	for (int i = 0; msg[i] != '\0'; ++i){
		printf("%c", msg[i]);
	}
	printf("\n");
}
